/*
* Copyright (c) 2022 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

import TestRunner from '@ohos.application.testRunner';
import AbilityDelegatorRegistry from '@ohos.app.ability.abilityDelegatorRegistry';
import { BusinessError } from '@ohos.base';

let abilityDelegator: AbilityDelegatorRegistry.AbilityDelegator;
let abilityDelegatorArguments: AbilityDelegatorRegistry.AbilityDelegatorArgs;

function translateParamsToString(parameters) {
  const keySet = new Set([
    '-s class', '-s notClass', '-s suite', '-s it',
    '-s level', '-s testType', '-s size', '-s timeout'
  ])
  let targetParams = '';
  for (const key in parameters) {
    if (keySet.has(key)) {
      targetParams = `${targetParams} ${key} ${parameters[key]}`;
    }
  }
  return targetParams.trim();
}

async function onAbilityCreateCallback() {
  console.log("onAbilityCreateCallback");
}

async function addAbilityMonitorCallback(err: BusinessError) {
  console.info("addAbilityMonitorCallback : " + JSON.stringify(err));
}

export default class OpenHarmonyTestRunner implements TestRunner {
  constructor() {
  }

  onPrepare() {
    console.info("OpenHarmonyTestRunner OnPrepare ");
  }

  async onRun() {
    console.log('OpenHarmonyTestRunner onRun run');
    abilityDelegatorArguments = AbilityDelegatorRegistry.getArguments();
    abilityDelegator = AbilityDelegatorRegistry.getAbilityDelegator();
    AppStorage.setOrCreate<AbilityDelegatorRegistry.AbilityDelegator>("abilityDelegator", abilityDelegator);
    let MainAbilityName = abilityDelegatorArguments.bundleName + '.MainAbility';
    let lMonitor = {
      abilityName: MainAbilityName,
      onAbilityCreate: onAbilityCreateCallback,
    }
    abilityDelegator.addAbilityMonitor(lMonitor, addAbilityMonitorCallback);
    let cmd = 'aa start -d 0 -a MainAbility' + ' -b ' + abilityDelegatorArguments.bundleName;
    cmd += ' ' + translateParamsToString(abilityDelegatorArguments.parameters);
    console.info('cmd : ' + cmd);
    abilityDelegator.executeShellCommand(cmd,
      (err: BusinessError, d: AbilityDelegatorRegistry.ShellCmdResult) => {
        console.info('executeShellCommand : err : ' + JSON.stringify(err));
        console.info('executeShellCommand : data : ' + d.stdResult);
        console.info('executeShellCommand : data : ' + d.exitCode);
      })
    console.info('OpenHarmonyTestRunner onRun end');
  }
}